LLM 强化学习中训练-推理失配的成因与修正
🧭 LLM 在线强化学习训练稳定性
George Washington University · Duke University
拆分数值偏差与策略滞后,提出 DLC 校正,稳定异步采样下 LLM 在线策略训练
📌 工作参考价值(3 点)
- 在线 RL 训练 LLM 推荐文案/对话 Agent 时,若推理引擎与 trainer 分离,先对每个 prefix 的 score 做 centering 再计算 advantage,可去除推理引擎数值差异带来的近固定偏差;类似 advantage normalization,但要按 prefix 而不是全局批量做。
- 对异步采样引入的 policy lag,不要只做 per-token importance weighting;持续监控重要性权重的 ESS,低于 0.5 时强制同步 sampler 权重。该阈值在 Qwen3-1.7B 上被验证为崩溃临界点,可作工程默认触发条件。
- 别只监控 sampler 端前缀上的 KL,它检测不到由滞后反馈导致的塌缩;建议增加与 on-policy baseline 的 KL 对比或 trainer 侧 KL 监控,并配置自动 resync。
策略梯度估计中的基线项与分数中心化
🧭 RLHF/策略梯度估计·重要性采样
University of Illinois Chicago · Perplexity AI
用 reward 与 importance-weighted score 的协方差替代正则化 IS 策略梯度,厘清 group-mean baseline 与 score centering 的不同作用
📌 工作参考价值(4 点)
- 在电商/Agent 的 LLM RLHF 后训练中,若用 sequence-level PPO/GRPO 且 ratio 被 clip,直接用 batch 内 mean reward 作为 group-mean baseline 即可,不必再对 importance-weighted score 做额外 centering,降低实现复杂度。
- 若采用 token-level advantage(细粒度 reward/PPO),当 importance ratio 被 truncation/masking/clipping 时,必须对 importance-weighted score 做 score centering(减去总体均值,可用 moving average 估计),否则策略梯度会带偏置。
- 建议把梯度写成 Cov(reward, weighted_score) 而非单纯加权 reward;该形式天然对 constant reward shift 稳健,适合离线异步训练中 stale policy 的纠偏。
- 序列级 IS 下 score centering 无额外收益,token 级 IS 下则是必要修正;实现前先确认 reward/advantage 的粒度再决定是否引入 score centering。
KL正则化策略优化:面向异步LLM Agent的单Rollout免Critic更新
🧭 LLM agent 强化学习策略优化
Princeton University
KLPO将KL正则锚定在采样器,实现每prompt单rollout、无critic的LLM agent策略优化
📌 工作参考价值(4 点)
- 在异步在线 RL(如导购 Agent、客服 Agent、广告文案优化)中,当采样器与训练器策略不一致时,避免 importance ratio clipping,改用 KLPO 的 sampler 锚定 + 最小二乘拟合,可减少 off-policy 偏差与实现复杂度。
- 每 prompt 单 rollout 且 critic-free,只依赖 terminal return,适合长轨迹/工具调用场景,省去 GRPO 成组采样成本,也省去 value 网络与 learned normalizer。
- 若业务已有 SPPO/GPO/REBEL/BPO 类策略优化,可参照 KLPO 统一视角进行替换或校准;top-K / binary KL 近似的精确 gap 公式可用于低成本估计 KL 项时的偏差权衡。
- 需注意论文为理论分析,未在具体推荐/电商数据上验证,实际落地前应先在小规模 agent 任务上验证稳定性与收益。
SANTA++:通过代表性键采样注意力以减少 KV 读取
🧭 LLM 长上下文推理 · 稀疏注意力采样 ★ 深度解读
University of California, Santa Barbara · Flucta
训练自由随机注意力方法,用代表性键路由团队采样,以 16%-22% KV 读取保留 94%-99% 精度
📌 工作参考价值(4 点)
- 在需要长上下文 LLM 推理(如用户长期行为序列、商品详情、多轮对话历史)的服务中,采用“代表性键 + 团队采样”做 decode 期稀疏 attention,可在 KV 读取降到 16%-25% 时仍保留接近 dense 的精度,适合降低在线推理成本。
- prefill 后一次性用 k-means 或 contiguous 分块构建 parent groups,再在每个 parent 内选 actual-key representatives(首点最近均值,后续逐次最远点),decode 阶段固定复用;避免逐 token 全量扫描,尤其在 batch-one 长序列下成本低。
- GQA 架构下,多个 query head 共享 KV head 的团队划分与代表键读取,但各 head 独立采样并保留自己的 inclusion correction,可进一步提升吞吐;与 MLA 等压缩 KV 表示互补,已在 DeepSeek-V2-Lite 上验证仍有 token 稀疏性。
- 实际部署注意:contiguous 分块可省去 k-means 预处理,虽然某些任务略降精度但访问更低;预算 S 是团队数而非 token 数,需要按团队大小和头间重叠重新估算实际 KV 读取。
分数中心化稳定离线策略强化学习
🧭 LLM 强化学习训练稳定性校正
Together AI
提出加性 score centering 校正项,抵消训练与推理引擎间的漂移,稳定 LLM 的 RL 训练
📌 工作参考价值(4 点)
- 在 RL 微调推荐/对话策略时,若部署推理引擎与训练引擎存在量化、KV cache 压缩或异步参数不一致,可在计算优势或回报时减去漂移项(如训练与推理引擎在同样本上的平均输出差异),低成本稳定训练,避免策略崩溃。
- Score centering 是加性修正,可与 importance sampling 等乘法修正叠加;在离线策略数据来自旧策略或不同推理引擎时,组合使用可能比单用 IS 更稳健,尤其 mismatch 严重时。
- 对于生成式推荐(用 LLM 直接生成 item ID/文案),RL 训练中 reward 常由规则或模型给出,容易受采样分布偏移影响;考虑在 reward 或 advantage 上做 centering 来抵消环境漂移,可提升训练稳定性。
- 工程上:漂移估计可通过在固定验证集上比较训练和推理引擎的 logprob/奖励差异得到,开销很小;可作为 RL 训练 pipeline 的常规监控与校正组件。
KV Cache 淘汰的概率视角:期望估计与解码期校正
🧭 LLM 推理 · KV cache 压缩 ★ 深度解读
University of California, Los Angeles
将 KV cache 淘汰形式化为期望估计问题,提出可纠偏的概率淘汰与解码期校正
📌 工作参考价值(4 点)
- 部署长上下文 LLM 服务(Agent 记忆、商品描述、用户行为序列)时,现有 H2O/SnapKV 等确定式 top-k 淘汰在特定任务上可能灾难性失效;改用论文的 probabilistic eviction + decode-time correction 能提升多任务鲁棒性,适合同一模型同时服务检索、问答、推荐解释等场景。
- 工程迁移成本低:可把已有 score-based 淘汰策略改造成 proposal 分布(如 πH2O=normalize H2O scores),按 head/group 采样并记录 counts,解码时用 Algorithm 1 的自归一化重要性采样校正,改动集中在 attention 计算,无需重训或改模型。
- 温度缩放 τ 是简单可用的 bias-variance 旋钮:高压缩、低采样数时增大 τ 抑制方差,低压缩时保持 τ=1 降低偏差;可按业务 latency/quality 目标做 per-head 或全局 budget 调优。
- 全局压缩率分配可利用论文的采样数与 coupon collector 关系做预算分配,让不同层/head 自适应获得不同压缩率;GQA 模型需用 group 内 mixture proposal 保持支持集一致。
核加权重要性采样用于上下文赌博机的离线策略评估
🧭 离线策略评估 · 重要性采样 ★ 深度解读
University College London · Strive Health Ltd. · National Institute for Health Research · Great Ormond Street Hospital · McGill University
将核平滑引入加权重要性采样,在行为策略误设时显著提升离线评估的准确性与稳健性
📌 工作参考价值(4 点)
- 当 logging policy 模型因策略更新而失准时,Kernel-WIS 提供更稳健的离线评估,适合推荐/广告系统的频繁策略迭代
- 核平滑思想可直接嵌入现有重要性采样评估流程:利用用户/物品的特征向量计算局部密度,对重要性权重做归一化,无需额外模型
- 带宽选择通过交叉验证自动完成,可部署为离线评估流程中的一个超参数调优模块,降低人工成本
- 连续奖励场景(如点击率预估)需谨慎使用,实验显示 Kernel-WIS 可能劣于 WIS,可考虑使用混合策略或对奖励离散化